package cn.huxingyuan.controller;

import cn.huxingyuan.articletype.domain.ArticleType;
import cn.huxingyuan.articletype.query.ArticleTypeQuery;
import cn.huxingyuan.service.IArticleTypeService;
import cn.huxingyuan.util.R;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;

/**
 * 文章分类管理
 * @author Hu
 * 2025/8/6 16:59
 */
@RestController
@RequestMapping("/articleType")
public class ArticleTypeController {
    @Autowired
    private IArticleTypeService articleTypeService;

    /**
     * 根据id获取文章分类
     * @param id id
     * @return 文章分类
     */
    @GetMapping("/{id}")
    public R findById(@PathVariable(value = "id") Long id) {
        ArticleType articleType = articleTypeService.findById(id);
        return R.ok(articleType);
    }

    /**
     * 查询全部
     * @return 全部文章分类
     */
    @GetMapping
    public R findAll() {
        List<ArticleType> all = articleTypeService.findAll();
        return R.ok(all);
    }

    /**
     * 保存文章分类信息（新增和修改）
     * @param articleType 文章分类信息
     * @return 统一返回类型信息
     */
    @PostMapping
    public R add(@RequestBody ArticleType articleType){
        System.out.println(articleType);
        if (articleType.getId() == null){
            articleTypeService.add(articleType);
        }else {
            if (articleType.getParentId() != articleType.getId()){
                articleTypeService.update(articleType);
            }else {
                return R.fail();
            }
        }
        return R.ok();
    }

    @DeleteMapping("/{id}")
    public R deleteById(@PathVariable(value = "id") Long id) {
        articleTypeService.deleteById(id);
        return R.ok();
    }

    @PostMapping("/batchDelete")
    public R batchDelete(@RequestBody List<Long> ids){
        articleTypeService.batchDelete(ids);
        return R.ok();
    }

    /**
     * 条件查询全部
     * @param articleTypeQuery 查询条件
     * @return 查询到的数据
     */
    @PostMapping("/pageAll")
    public R pageAll(@RequestBody ArticleTypeQuery articleTypeQuery) {
        return R.ok(articleTypeService.pageAll(articleTypeQuery));
    }
    @GetMapping("/getTree")
    public R getTree(){
        //查询全部的文章分类
        List<ArticleType> all = articleTypeService.findAll();
        //装在map里方便分出找到子类
        HashMap<Long, ArticleType> map = new HashMap<>();
        for (ArticleType item : all){
                map.put(item.getId(),item);
        }

        ArrayList<ArticleType> tree = new ArrayList<>();

        for (ArticleType articleType : all){
            Long parentId = articleType.getParentId();
            if (parentId == null){
                tree.add(articleType);
            }else {
                map.get(parentId).getChildren().add(articleType);
            }
        }
        return R.ok(tree);
    }
}
